# test_text2cypher.py

import pytest
from mcp_server_graph.text2cypher_agent import text2cypher


@pytest.mark.asyncio
async def test_user_car_fam():
    final = await text2cypher("北京市有购车意向的用户")
    print(final["interpretation"])
    assert len(final["interpretation"]) > 0


@pytest.mark.asyncio
async def test_new_customer():
    final = await text2cypher("北京市异网拉新的案例")
    print(final["interpretation"])
    assert len(final["interpretation"]) > 0

@pytest.mark.asyncio
async def test_group_operation():
    final = await text2cypher("北京市群体运营")
    print(final["interpretation"])
    assert len(final["interpretation"]) > 0

@pytest.mark.asyncio
async def test_group_prodct_recomand():
    final = await text2cypher("根据个人用户产品订购的情况，发掘产品的潜在合作企业")
    print(final["interpretation"])
    assert len(final["interpretation"]) > 0

@pytest.mark.asyncio
async def test_fraud():
    final = await text2cypher("北京市疑似诈骗的用户案例")
    print(final["interpretation"])
    assert len(final["interpretation"]) > 0